Risk-Controlled KV-Cache Eviction: From Memory Budgets to Risk Targets
风险受控的 KV Cache 淘汰:从内存预算到风险目标
🧭 LLM 推理 · KV cache 风险控制
Department of Artificial Intelligence, Korea University
将 KV cache 淘汰重新定义为部署风险控制,用事后认证选择保留策略并按需回退全 KV
📌 工作参考价值(4 点)
- 在电商/广告线上长上下文 LLM 服务(商品详情问答、评论总结、Agent 长时记忆)中,用风险预算替代平均精度来选择 KV cache 压缩策略:按业务可容忍的退化概率(如 5%)和置信度进行校准,只上线被认证的策略,避免长尾请求质量事故。
- 引入全 KV fallback 机制:当校准集上没有任何压缩策略通过风险认证(例如高难度长上下文任务),自动回退到不压缩,保障最坏情况体验,可作为线上安全网。
- 离线评测时不要只看平均退化率,要报告有限样本认证的风险上界;对于固定内存预算方法,认证可能显示需要多保留 5-10 个百分点的 KV cache,帮助更稳健地设置内存配额。
- 该方法是 compressor-agnostic 的,可与任意 KV 淘汰算法结合,适合在现有推理加速框架上低成本包装为上线前认证流程。